/*
 * @Descripttion:
 * @version:
 * @Author: xiaozilai
 * @Date: 2022-11-25 18:39:17
 * @LastEditors: xiaozilai
 * @LastEditTime: 2022-11-25 18:51:25
 */
/*
 * @lc app=leetcode.cn id=809 lang=cpp
 *
 * [809] 情感丰富的文字
 *
 * https://leetcode.cn/problems/expressive-words/description/
 *
 * algorithms
 * Medium (41.64%)
 * Likes:    101
 * Dislikes: 0
 * Total Accepted:    18.4K
 * Total Submissions: 39K
 * Testcase Example:  '"heeellooo"\n["hello", "hi", "helo"]'
 *
 * 有时候人们会用重复写一些字母来表示额外的感受，比如 "hello" -> "heeellooo", "hi" ->
 * "hiii"。我们将相邻字母都相同的一串字符定义为相同字母组，例如："h", "eee", "ll", "ooo"。
 *
 * 对于一个给定的字符串 S ，如果另一个单词能够通过将一些字母组扩张从而使其和 S
 * 相同，我们将这个单词定义为可扩张的（stretchy）。扩张操作定义如下：选择一个字母组（包含字母 c ），然后往其中添加相同的字母 c 使其长度达到
 * 3 或以上。
 *
 * 例如，以 "hello" 为例，我们可以对字母组 "o" 扩张得到 "hellooo"，但是无法以同样的方法得到 "helloo" 因为字母组 "oo"
 * 长度小于 3。此外，我们可以进行另一种扩张 "ll" -> "lllll" 以获得 "helllllooo"。如果 s =
 * "helllllooo"，那么查询词 "hello" 是可扩张的，因为可以对它执行这两种扩张操作使得 query = "hello" ->
 * "hellooo" -> "helllllooo" = s。
 *
 * 输入一组查询单词，输出其中可扩张的单词数量。
 *
 *
 *
 * 示例：
 *
 *
 * 输入：
 * s = "heeellooo"
 * words = ["hello", "hi", "helo"]
 * 输出：1
 * 解释：
 * 我们能通过扩张 "hello" 的 "e" 和 "o" 来得到 "heeellooo"。
 * 我们不能通过扩张 "helo" 来得到 "heeellooo" 因为 "ll" 的长度小于 3 。
 *
 *
 *
 *
 * 提示：
 *
 *
 * 1 <= s.length, words.length <= 100
 * 1 <= words[i].length <= 100
 * s 和所有在 words 中的单词都只由小写字母组成。
 *
 *
 */

// @lc code=start
#include <iostream>
#include <vector>
using namespace std;
class Solution {
    vector<pair<char, int>> dict;

public:
    int expressiveWords(string s, vector<string> &words) {
        dict = count(s);
        int ans = 0;
        for (auto &word : words) {
            if (isStretchy(word)) {
                ++ans;
            }
        }
        return ans;
    }

    vector<pair<char, int>> count(string &s) {
        vector<pair<char, int>> ret;
        int length = s.length();
        for (int i = 0; i < length; ++i) {
            if (i > 0 && s[i] == s[i - 1]) {
                ++ret.back().second;
            } else {
                ret.push_back({s[i], 1});
            }
        }
        return ret;
    }

    bool isStretchy(string &word) {
        vector<pair<char, int>> temp = count(word);
        if (temp.size() != dict.size()) {
            return false;
        }
        int n = temp.size();
        for (int i = 0; i < n; ++i) {
            if (dict[i].first != temp[i].first || dict[i].second < temp[i].second || (dict[i].second != temp[i].second && dict[i].second < 3)) {
                return false;
            }
        }
        return true;
    }
};
// @lc code=end
